package leetcode.editor.cn;

import leetcode.editor.cn.common.ListNode;

/**
 * @Author: Dempsey
 * @Data: 2021-01-05 19:01:07
 */

//给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。 
//
// k 是一个正整数，它的值小于或等于链表的长度。 
//
// 如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。 
//
// 
//
// 示例： 
//
// 给你这个链表：1->2->3->4->5 
//
// 当 k = 2 时，应当返回: 2->1->4->3->5 
//
// 当 k = 3 时，应当返回: 3->2->1->4->5 
//
// 
//
// 说明： 
//
// 
// 你的算法只能使用常数的额外空间。 
// 你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。 
// 
// Related Topics 链表 
// 👍 833 👎 0


public class P25 {
    public static void main(String[] args) {
        Solution solution = new P25().new Solution();
    }

//leetcode submit region begin(Prohibit modification and deletion)

    /**
     * Definition for singly-linked list.
     * public class ListNode {
     * int val;
     * ListNode next;
     * ListNode() {}
     * ListNode(int val) { this.val = val; }
     * ListNode(int val, ListNode next) { this.val = val; this.next = next; }
     * }
     */
    class Solution {
        public ListNode reverseKGroup(ListNode head, int k) {
            if (k == 1 || head == null) return head;
            return helper(head, k - 1);
        }

        public ListNode helper(ListNode head, int k) {
            if (head == null) return null;
            ListNode start = head, end = start;
            int index;
            //让end移动到第k个节点上
            for (index = 0; index < k; ++index) {
                if (end == null) break;
                end = end.next;
            }
            //这种情况下，说明不够k个，直接返回head，即不翻转
            if (index != k || end == null) return head;
            ListNode node = end == null ? null : end.next;
            //这里先将k个节点与后面的连接断开，方便翻转链表
            end.next = null;
            //翻转这k个节点
            reverse(start);
            //让前k个节点的头节点的next指向后k个节点的尾节点
            start.next = helper(node, k);
            return end;
        }

        /**
         * 翻转链表
         *
         * @param head
         */
        public ListNode reverse(ListNode head) {
            ListNode pre = head, next = pre.next;
            while (next != null) {
                ListNode tmp = next.next;
                next.next = pre;
                pre = next;
                next = tmp;
            }
            return head;
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}